from pydantic import BaseModel
from typing import List, Optional

# 注册请求
class RegisterRequest(BaseModel):
    email: str
    password: str

# 注册响应
class RegisterResponse(BaseModel):
    success: bool
    message: Optional[str] = None

# 登录请求
class LoginRequest(BaseModel):
    email: str
    password: str

# 登录响应
class TokenResponse(BaseModel):
    access_token: str | None = None
    token_type: str | None = None
    message: str | None = None

class UserInfo(BaseModel):
    email: str
    vip: bool
    disk_quota: int
    disk_usage: int
